The Atari Compendium/The Atari Compendium (Toad Computers) (1994).iso/files/prgtools/mint/utilit~1/futilsrc.zoo/fileutil/src/mvdir.c

C/C++ Source or Header | 1991-08-21 | 4.8 KB | 172 lines

/* mvdir -- rename directory
   Copyright (C) 1990-1991 Free Software Foundation, Inc.

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2, or (at your option)
   any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */

/* Helper program for GNU mv on machines that lack the rename system call.

   Usage: mvdir from to

   FROM must be an existing directory.
   TO must not exist, but its parent must exist.

   Must be setuid root.

   Ian Dall ([e-mail])
   and David MacKenzie ([e-mail]) */

#include <stdio.h>
#include <sys/types.h>
#include <signal.h>
#include "system.h"

#ifndef HIPRI
#define HIPRI -10
#endif

#ifdef DEBUG
#define link(FROM, TO) (printf("Linking %s to %s\n", FROM, TO), 0)
#define unlink(FILE) (printf("Unlinking %s\n", FILE), 0)
#endif

/* The name this program was run with. */
char *program_name;

char *basename ();
char *fullpath ();
char *dirname ();
char *xmalloc ();
void error ();
void strip_trailing_slashes ();

void
main (argc, argv)
     int argc;
     char **argv;
{
  char *from, *from_parent, *from_base;
  char *to, *to_parent, *to_parent_path;
  struct stat from_stats, to_stats;
  char *slash, temp;
  int last_elem;
  int i;

  program_name = argv[0];
  if (argc != 3)
    {
      fprintf (stderr, "Usage: %s existing-dir new-dir\n", program_name);
      exit (2);
    }
  from = argv[1];
  to = argv[2];
  strip_trailing_slashes (from);
  strip_trailing_slashes (to);
  from_parent = dirname (from);
  to_parent = dirname (to);
  if (!from_parent || !to_parent)
    error (2, 0, "virtual memory exhausted");

  /* Make sure `from' is not "." or "..". */
  from_base = basename (from);
  if (!strcmp (from_base, ".") || !strcmp (from_base, ".."))
    error (1, 0, "cannot rename `.' or `..'");

  /* Even with an effective uid of root, link fails if the target exists.
     That is what we want, so don't unlink `to' first.
     However, we do need to check that the directories that link and unlink
     will modify exist and are writable by the user. */

  if (stat (from, &from_stats))
    error (1, errno, "%s", from);
  if (!S_ISDIR (from_stats.st_mode))
    error (1, 0, "`%s' is not a directory", from);
  if (access (from_parent, W_OK))
    error (1, errno, "cannot write to `%s'", from_parent);
  if (access (to_parent, W_OK))
    error (1, errno, "cannot write to `%s'", to_parent);

  /* To prevent disconnecting the tree rooted at `from' from its parent,
     quit if any of the directories in `to' are the same (dev and ino)
     as the directory `from'. */

  slash = to_parent_path = fullpath (to_parent);
  last_elem = 0;
  while (1)
    {
      while (*slash == '/')
    ++slash;
      slash = index (slash, '/');
      if (slash != NULL)
    *slash = '\0';
      else
    last_elem = 1;
      if (stat (to_parent_path, &to_stats))
    error (1, errno, "%s", to_parent_path);
      if (to_stats.st_dev == from_stats.st_dev
      && to_stats.st_ino == from_stats.st_ino)
    error (1, 0, "`%s' is an ancestor of `%s'", from, to);
      if (last_elem)
    break;
      *slash++ = '/';
    }

  /* We can't make the renaming atomic, but we do our best. */
  for (i = NSIG; i > 0; i--)
    if (i != SIGKILL)
      signal (i, SIG_IGN);
  setuid (0);            /* Make real uid 0 so it is harder to kill. */
  nice (HIPRI - nice (0));    /* Raise priority. */

  if (link (from, to))
    error (1, errno, "cannot link `%s' to `%s'", from, to);
  if (unlink (from))
    error (1, errno, "cannot unlink `%s'", from);

  /* Replace the directory's `..' entry.  It used to be a link to
     the parent of `from'; make it a link to the parent of `to' instead. */
  i = strlen (to);
  slash = xmalloc (i + 4);
  strcpy (slash, to);
  strcpy (slash + i, "/..");
  if (unlink (slash) && errno != ENOENT)
    error (1, errno, "cannot unlink `%s'", slash);
  if (link (to_parent, slash))
    error (1, errno, "cannot link `%s' to `%s'", to_parent, slash);

  exit (0);
}

/* Return the full pathname (from /) of the directory DIR,
   as static data. */

char *
fullpath (dir)
     char *dir;
{
  char wd[PATH_MAX + 2];
  static char path[PATH_MAX + 2];

  if (getwd (wd) == NULL)
    error (1, errno, "cannot get current directory");
  if (chdir (dir))
    error (1, errno, "%s", dir);
  if (getwd (path) == NULL)
    error (1, errno, "cannot get current directory");
  if (chdir (wd))
    error (1, errno, "%s", wd);

  return path;
}